iT邦幫忙

第 11 屆 iThome 鐵人賽

DAY 17
0
AI & Data

AI+Line系列 第 17

Day17 CNN卷積神經網路程式(5):儲存模型、讀取模型

  • 分享至 

  • xImage
  •  

好不容易訓練好的模型,可別一關閉程式就消失了,儲存、讀取的做法十分簡單

儲存模型:
將訓練好的模型存起來,可以將來繼續訓練、或是讀取使用
存成HDF5檔案, 如果沒有這個套件的話,請pip install h5py
model.save('./model.h5')

讀取模型:
model = load_model('./model.h5')

在Heroku伺服器,我們用的就是讀取已經訓練好的模型,省去訓練的時間

def init_default(self):
    self.model = Sequential()
    """ 
    建構一樣的模型... 
    """

    del self.model  # deletes the existing model
    self.model = load_model('./model/number_model.h5')
    self.graph = tf.get_default_graph()
    print('Reloaded')

上一篇
Day16 CNN卷積神經網路程式(4):訓練模型
下一篇
Day18 CNN卷積神經網路程式(6):預測新資料
系列文
AI+Line30
圖片
  直播研討會
圖片
{{ item.channelVendor }} {{ item.webinarstarted }} |
{{ formatDate(item.duration) }}
直播中

尚未有邦友留言

立即登入留言